17905f56c9c697dd3b0c34cfbc184adb3d643af8,modules/assignment/src/test/java/com/intuit/wasabi/assignment/impl/AssignmentsImplTest.java,AssignmentsImplTest,getAssignment_test_4,#,840
Before Change
Mockito.when(assignmentsRepository.assignUser(assignment, experiment, DATE)).thenReturn(newAssignment);
Mockito.when(assignmentsRepository.getAssignment(experiment.getID(), User.ID.valueOf("user-b"), context)).thenReturn(newAssignment);
assignment = assignmentsRepository.getAssignment(experiment.getID(), User.ID.valueOf("user-b"), context);
assertTrue(assignment.getBucketLabel().equals(experienceB.getLabel()));
assertTrue(assignment.getStatus() == Assignment.Status.NEW_ASSIGNMENT);
newAssignment = Assignment.newInstance(assignment.getExperimentID())
.withBucketLabel(assignment.getBucketLabel())
.withUserID(assignment.getUserID())
.withContext(assignment.getContext())
.withStatus(Assignment.Status.EXISTING_ASSIGNMENT)
.withCreated(DATE)
.withCacheable(null)
.build();
Mockito.when(assignmentsRepository.getAssignment(experiment.getID(), User.ID.valueOf("user-b"), context)).thenReturn(newAssignment);
assignment = assignmentsRepository.getAssignment(experiment.getID(), User.ID.valueOf("user-b"), context);
assertTrue(assignment.getStatus() == Assignment.Status.EXISTING_ASSIGNMENT);
assertTrue(assignment.getBucketLabel().equals(experienceB.getLabel()));
assertTrue(assignment.getStatus() == Assignment.Status.EXISTING_ASSIGNMENT);
builder = Assignment.newInstance(experiment.getID())
.withApplicationName(experiment.getApplicationName())
.withUserID(User.ID.valueOf("user-c"))
.withContext(context)
.withBucketLabel(experienceC.getLabel());
assignment = builder.build();
newAssignment = Assignment.newInstance(assignment.getExperimentID())
.withBucketLabel(assignment.getBucketLabel())
.withUserID(assignment.getUserID())
.withContext(assignment.getContext())
.withStatus(Assignment.Status.NEW_ASSIGNMENT)
.withCreated(DATE)
.withCacheable(null)
.build();
Mockito.when(assignmentsRepository.assignUser(assignment, experiment, DATE)).thenReturn(newAssignment);
Mockito.when(assignmentsRepository.getAssignment(experiment.getID(), User.ID.valueOf("user-c"), context)).thenReturn(newAssignment);
assignment = assignmentsRepository.getAssignment(experiment.getID(), User.ID.valueOf("user-c"), context);
assertTrue(assignment.getBucketLabel() == experienceC.getLabel());
assertTrue(assignment.getStatus() == Assignment.Status.NEW_ASSIGNMENT);
After Change
.withCacheable(null)
.build();
Mockito.when(assignmentsRepository.getAssignment(User.ID.valueOf("user-b"), testApp, experiment.getID(), context)).thenReturn(newAssignment);
assignment = assignmentsRepository.getAssignment(User.ID.valueOf("user-b"), testApp, experiment.getID(), context);
assertTrue(assignment.getStatus() == Assignment.Status.EXISTING_ASSIGNMENT);
assertTrue(assignment.getBucketLabel().equals(experienceB.getLabel()));
assertTrue(assignment.getStatus() == Assignment.Status.EXISTING_ASSIGNMENT);
builder = Assignment.newInstance(experiment.getID())
.withApplicationName(experiment.getApplicationName())
.withUserID(User.ID.valueOf("user-c"))
.withContext(context)
.withBucketLabel(experienceC.getLabel());
assignment = builder.build();
newAssignment = Assignment.newInstance(assignment.getExperimentID())
.withBucketLabel(assignment.getBucketLabel())
.withUserID(assignment.getUserID())
.withContext(assignment.getContext())
.withStatus(Assignment.Status.NEW_ASSIGNMENT)
.withCreated(DATE)
.withCacheable(null)
.build();
Mockito.when(assignmentsRepository.getAssignment(User.ID.valueOf("user-c"), testApp, experiment.getID(), context)).thenReturn(newAssignment);
assignment = assignmentsRepository.getAssignment(User.ID.valueOf("user-c"), testApp, experiment.getID(), context);
assertTrue(assignment.getBucketLabel() == experienceC.getLabel());